from .cross_entropy import FusedCrossEntropy